package com.jnc.plug.mybatis.mp.generator.config;

import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.GlobalConfig;

import java.io.File;

/**
 * @Description
 * @Author fanlei
 * @Version 1.0
 * @Date 2019/9/28
 */
public class CustomGlobalConfig {
    private AutoGenerator mpg;

    public CustomGlobalConfig(AutoGenerator mpg) {
        this.mpg = mpg;
    }

    public void invoke() {
        mpg.setGlobalConfig(new GlobalConfig()
                .setOutputDir(Context.outputDir)//输出目录
                .setFileOverride(true)// 是否覆盖文件
                .setActiveRecord(false)// 开启 activeRecord 模式
                .setEnableCache(false)// XML 二级缓存
                .setBaseResultMap(Context.baseResultMap)// XML ResultMap
                .setBaseColumnList(true)// XML columList
                .setSwagger2(Context.swagger2)
                // 自定义文件命名，注意 %s 会自动填充表实体属性！
                .setEntityName(Context.entityName)
                .setControllerName(Context.controllerName)
                .setServiceName(Context.serviceName)
                .setServiceImplName(Context.serviceImplName)
                .setMapperName(Context.mapperName)
                .setXmlName(Context.xmlName)
                //.setKotlin(true) 是否生成 kotlin 代码
                .setAuthor(Context.author));
    }
}
